package com.luobr.backend.manage.upload;

import cn.hutool.core.date.DateUtil;
import cn.hutool.core.io.FileUtil;
import cn.hutool.core.util.RandomUtil;
import cn.hutool.core.util.StrUtil;
import com.luobr.backend.api.imageInfo.ImageInfoGetApi;
import com.luobr.backend.common.ResponseCode;
import com.luobr.backend.config.OssClientConfig;
import com.luobr.backend.exception.BusinessException;
import com.luobr.backend.manage.OssManager;
import com.luobr.backend.model.dto.image.ImageInfoResult;
import com.luobr.backend.model.dto.image.ImageParseResult;
import lombok.extern.slf4j.Slf4j;

import javax.annotation.Resource;
import java.io.File;
import java.util.Date;

/**
 * @author: 罗念笙
 * @date: 2025/3/2 14:55
 */
@Slf4j
public abstract class UploadAndParseImageTemplate {

    @Resource
    private OssManager ossManager;
    @Resource
    private OssClientConfig ossClientConfig;

    /**
     * 校验图片
     * @param inputSource 输入源
     */
    public abstract void validImage(Object inputSource);

    /**
     * 获取原来的文件名
     * @param inputSource 输入源
     * @return
     */
    public abstract String getOriginalFilename(Object inputSource);

    /**
     * 处理文件
     * @param inputSource 输入源
     * @param file 临时文件
     */
    public abstract void processFile(Object inputSource, File file);

    /**
     * 上传图片 + 解析图片
     *
     * @param inputSource 输入源
     * @param uploadPathPrefix 上传路径前缀，例如：test/xxx.png 中的 test
     * @return 返回图片解析之后的信息
     */
    public ImageParseResult uploadAndParseImage(Object inputSource, String uploadPathPrefix) {
        // 1、校验图片
        validImage(inputSource);
        // 2、图片上传地址（拼接图片的名称以及相对路径）
        String uuid = RandomUtil.randomString(16); // 16位随机数
        String originFilename = getOriginalFilename(inputSource);
        String suffixName = StrUtil.isBlank(FileUtil.getSuffix(originFilename)) ? "jpg" : FileUtil.getSuffix(originFilename); // 文件后缀名
        String uploadFileName = String.format("%s_%s.%s", DateUtil.formatDate(new Date()), uuid, suffixName); // 上传图片的名称（保证唯一）
        String uploadPath = String.format("%s/%s", uploadPathPrefix, uploadFileName); // 上传图片的相对路径，例如 test/xxx.png

        File file = null;
        try {
            // 3、创建临时文件和处理临时文件
            file = File.createTempFile(uploadPath, null);
            processFile(inputSource,file);
            // 4、上传图片到 OSS 对象存储过程
            ossManager.putObject(uploadPath, file);
            // 5、获取图片信息
            ImageInfoResult imageInfoResult = ossManager.getImageObject(uploadPath);
            ImageParseResult imageParseResult = ImageParseResult.imageInfoToParseVo(imageInfoResult);
            // 图片名
            imageParseResult.setName(FileUtil.mainName(originFilename));
            // 图片路径
            String url = String.format("%s.%s/%s", "https://" + ossClientConfig.getBucketName(), ossClientConfig.getEndpoint(), uploadPath);
            imageParseResult.setUrl(url);
            // 图片主色调
            String imgColor = ImageInfoGetApi.getImageColor(url);
            imageParseResult.setImgColor(imgColor);
            return imageParseResult;
        } catch (Exception e) {
            log.error("图片上传到对象存储失败", e);
            throw new BusinessException(ResponseCode.SYSTEM_ERROR, "上传失败");
        } finally {
            this.deleteTempFile(file);
        }
    }

    /**
     * 删除临时文件
     */
    public void deleteTempFile(File file) {
        if (file == null) {
            return;
        }
        // 删除临时文件
        if (!file.delete()) {
            log.error("file delete error, filePath = {}", file.getAbsolutePath());
        }
    }
}
